package com.community.common.util;

import com.community.common.config.BizException;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.*;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

/**
 * @author zhu
 * @Desc: excel处理类
 */
public class ExcelUtil {

    public static List<Map<String, String>> getData(String tableName, String[] columns) {
        return getData(0, tableName, columns);
    }

    public static List<Map<String, String>> getData(int page, String tableName, String[] columns) {
        Sheet sheet = null;
        Row row = null;
        String cellData = null;
        String filePath = "D:" + File.separator + tableName;
        Workbook wb = readExcel(filePath);
        if (wb == null) {
            throw new BizException("读取EXCEL表格失败");
        }
        //用来存放表中数据
        List<Map<String, String>> list = new ArrayList<>();
        //获取第一个sheet
        sheet = wb.getSheetAt(page);
        //获取最大行数
        int rownum = sheet.getPhysicalNumberOfRows();
        //获取标题列
        row = sheet.getRow(0);
        //获取最大列数
        int colnum = row.getPhysicalNumberOfCells();
        for (int i = 1; i < rownum; i++) {
            Map<String, String> map = new LinkedHashMap<>();
            row = sheet.getRow(i);
            if (row != null) {
                for (int j = 0; j < colnum; j++) {
                    cellData = getCellFormatValue(row.getCell(j));
                    map.put(columns[j], cellData);
                }
            } else {
                break;
            }
            list.add(map);
        }
        return list;
    }

    //读取excel
    private static Workbook readExcel(String filePath) {
        Workbook wb = null;
        if (filePath == null) {
            return null;
        }
        String extString = filePath.substring(filePath.lastIndexOf("."));
        InputStream is = null;
        try {
            is = new FileInputStream(filePath);
            if (".xls".equals(extString)) {
                return wb = new HSSFWorkbook(is);
            } else if (".xlsx".equals(extString)) {
                return wb = new XSSFWorkbook(is);
            } else {
                return wb = null;
            }
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return wb;
    }

    private static String getCellFormatValue(Cell cell) {
        String cellValue = null;
        if (cell != null) {
            //判断cell类型
            switch (cell.getCellType()) {
                case Cell.CELL_TYPE_NUMERIC: {
                    cellValue = String.valueOf(cell.getNumericCellValue());
                    break;
                }
                case Cell.CELL_TYPE_STRING: {
                    cellValue = cell.getRichStringCellValue().getString();
                    break;
                }
                default:
                    cellValue = "";
            }
        } else {
            cellValue = "";
        }
        return cellValue;
    }
}
